package top.lucky.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;

/**
 * AES加解密工具类
 */
@Slf4j
public class AesNewUtil {
    // 算法定义
    private static final String AES_ALGORITHM = "AES";
    // 指定填充方式
    private static final String AES_PADDING = "AES/CBC/PKCS5Padding";

    /***
     * AES加密
     *
     * @param content 待加密内容
     * @param aesKey 密钥
     * @return
     */
    public static String encrypt(String content, String aesKey) throws Exception{
        if(StringUtils.isBlank(content)){
            return null;
        }
        // 默认使用U8编码
        return encrypt(content, aesKey, StandardCharsets.UTF_8);
    }

    /***
     * AES加密
     *
     * @param content 待加密内容
     * @param aesKey 密钥
     * @param encoding 编码
     * @return
     */
    public static String encrypt(String content, String aesKey, Charset encoding) throws Exception {
        try {
            IvParameterSpec zeroIv = new IvParameterSpec(aesKey.getBytes());
            SecretKeySpec key = new SecretKeySpec(aesKey.getBytes(encoding), AES_ALGORITHM);
            Cipher cipher = Cipher.getInstance(AES_PADDING);// 创建密码器
            byte[] byteContent = content.getBytes(encoding);
            cipher.init(Cipher.ENCRYPT_MODE, key, zeroIv);// 初始化
            // 加密
            byte[] result = cipher.doFinal(byteContent);
            // 转换16进制字符串后返回
            return ByteUtil.bytesToHexString(result);
        } catch (Exception e) {
            log.error("aes 加密失败", e);
            throw new Exception(e);
        }
    }

    /***
     * AES解密，输入密文16进制字符串,默认使用U8编码
     *
     * @param cipherText
     * @param aesKey
     * @return
     */
    public static String decrypt(String cipherText, String aesKey) throws Exception{
        // 将16进制的密文转换为byte类型
        byte[] cipherTextByte = ByteUtil.hexStringToBytes(cipherText);
        if (cipherTextByte == null) {
            return null;
        }
        // 默认使用U8编码返回原字符串
        return decrypt(cipherTextByte, aesKey, StandardCharsets.UTF_8);
    }

    /***
     * AES解密，输入为byte数组,默认使用U8编码
     *
     * @param cipherTextByte 加密内容
     * @param aesKey 密钥
     * @return
     */
    public static String decrypt(byte[] cipherTextByte, String aesKey) throws Exception{
        // 默认使用U8编码返回原字符串
        return decrypt(cipherTextByte, aesKey, StandardCharsets.UTF_8);
    }

    /***
     * AES解密，输入为byte数组,可以指定编码
     *
     * @param cipherTextByte 加密内容
     * @param aesKey 密钥
     * @param encoding 编码
     * @return
     */
    public static String decrypt(byte[] cipherTextByte, String aesKey, Charset encoding) throws Exception {
        try {
            IvParameterSpec zeroIv = new IvParameterSpec(aesKey.getBytes(StandardCharsets.UTF_8));
            SecretKeySpec key = new SecretKeySpec(aesKey.getBytes(encoding), AES_ALGORITHM);
            Cipher cipher = Cipher.getInstance(AES_PADDING);// 创建密码器
            cipher.init(Cipher.DECRYPT_MODE, key, zeroIv);// 初始化
            byte[] result = cipher.doFinal(cipherTextByte);
            return new String(result, encoding);
        } catch (Exception e) {
            log.error("aes 解密失败", e);
            throw new Exception(e);
        }
    }

}